# AUTHOR: DING
# -*- codeing = utf-8 -*-
# @Time: 2024/2/29 10:37
# @Author: 86139
# @Site: 
# @File: 20-models.py
# @Software: PyCharm
# tensorboard --logdir=pytorch/logs --port=6007
import torch
import torch.nn as nn
import torchvision.datasets
import torchvision

vgg16_true = torchvision.models.vgg16(pretrained=True)
vgg16_false = torchvision.models.vgg16(pretrained=False)
# print(vgg16_true)
dataset = torchvision.datasets.CIFAR10("./dataset", train=False, transform=torchvision.transforms.ToTensor(),
                                       download=True)
vgg16_true.add_module("add_linear", nn.Linear(1000, 10))
print(vgg16_true)